<?php

if (!defined('BASEPATH'))
  exit('No direct script access allowed');
/**
 * CodeIgniter
 *
 * An open source application development framework for PHP 5.1.6 or newer
 *
 * @package		CodeIgniter
 * @author		ExpressionEngine Dev Team
 * @copyright	Copyright (c) 2008 - 2011, EllisLab, Inc.
 * @license		http://codeigniter.com/user_guide/license.html
 * @link		http://codeigniter.com
 * @since		Version 1.0
 * @filesource
 */
// ------------------------------------------------------------------------

/**
 * Zip Compression Class
 *
 * This class is based on a library I found at Zend:
 * http://www.zend.com/codex.php?id=696&single=1
 *
 * The original library is a little rough around the edges so I
 * refactored it and added several additional methods -- Rick Ellis
 *
 * @package		CodeIgniter
 * @subpackage	Libraries
 * @category	Encryption
 * @author		ExpressionEngine Dev Team
 * @link		http://codeigniter.com/user_guide/libraries/zip.html
 */
class CI_Zip {

  var $zipdata = '';
  var $directory = '';
  var $entries = 0;
  var $file_num = 0;
  var $offset = 0;
  var $now;

  /**
   * Constructor
   */
  public function __construct() {
    log_message('debug', "Zip Compression Class Initialized");

    $this->now = time();
  }

  // --------------------------------------------------------------------

  /**
   * Add Directory
   *
   * Lets you add a virtual directory into which you can place files.
   *
   * @access	public
   * @param	mixed	the directory name. Can be string or array
   * @return	void
   */
  function add_dir($directory) {
    foreach ((array) $directory as $dir) {
      if (!preg_match("|.+/$|", $dir)) {
        $dir .= '/';
      }

      $dir_time = $this->_get_mod_time($dir);

      $this->_add_dir($dir, $dir_time['file_mtime'], $dir_time['file_mdate']);
    }
  }

  // --------------------------------------------------------------------

  /**
   * 	Get file/directory modification time
   *
   * 	If this is a newly created file/dir, we will set the time to 'now'
   *
   * 	@param string	path to file
   * 	@return array	filemtime/filemdate
   */
  function _get_mod_time($dir) {
    // filemtime() will return false, but it does raise an error.
    $date = (@filemtime($dir)) ? filemtime($dir) : getdate($this->now);

    $time['file_mtime'] = ($date['hours'] << 11) + ($date['minutes'] << 5) + $date['seconds'] / 2;
    $time['file_mdate'] = (($date['year'] - 1980) << 9) + ($date['mon'] << 5) + $date['mday'];

    return $time;
  }

  // --------------------------------------------------------------------

  /**
   * Add Directory
   *
   * @access	private
   * @param	string	the directory name
   * @return	void
   */
  function _add_dir($dir, $file_mtime, $file_mdate) {
    $dir = str_replace("\\", "/", $dir);

    $this->zipdata .=
            "\x50\x4b\x03\x04\x0a\x00\x00\x00\x00\x00"
            . pack('v', $file_mtime)
            . pack('v', $file_mdate)
            . pack('V', 0) // crc32
            . pack('V', 0) // compressed filesize
            . pack('V', 0) // uncompressed filesize
            . pack('v', strlen($dir)) // length of pathname
            . pack('v', 0) // extra field length
            . $dir
            // below is "data descriptor" segment
            . pack('V', 0) // crc32
            . pack('V', 0) // compressed filesize
            . pack('V', 0); // uncompressed filesize

    $this->directory .=
            "\x50\x4b\x01\x02\x00\x00\x0a\x00\x00\x00\x00\x00"
            . pack('v', $file_mtime)
            . pack('v', $file_mdate)
            . pack('V', 0) // crc32
            . pack('V', 0) // compressed filesize
            . pack('V', 0) // uncompressed filesize
            . pack('v', strlen($dir)) // length of pathname
            . pack('v', 0) // extra field length
            . pack('v', 0) // file comment length
            . pack('v', 0) // disk number start
            . pack('v', 0) // internal file attributes
            . pack('V', 16) // external file attributes - 'directory' bit set
            . pack('V', $this->offset) // relative offset of local header
            . $dir;

    $this->offset = strlen($this->zipdata);
    $this->entries++;
  }

  // --------------------------------------------------------------------

  /**
   * Add Data to Zip
   *
   * Lets you add files to the archive. If the path is included
   * in the filename it will be placed within a directory.  Make
   * sure you use add_dir() first to create the folder.
   *
   * @access	public
   * @param	mixed
   * @param	string
   * @return	void
   */
  function add_data($filepath, $data = NULL) {
    if (is_array($filepath)) {
      foreach ($filepath as $path => $data) {
        $file_data = $this->_get_mod_time($path);

        $this->_add_data($path, $data, $file_data['file_mtime'], $file_data['file_mdate']);
      }
    } else {
      $file_data = $this->_get_mod_time($filepath);

      $this->_add_data($filepath, $data, $file_data['file_mtime'], $file_data['file_mdate']);
    }
  }

  // --------------------------------------------------------------------

  /**
   * Add Data to Zip
   *
   * @access	private
   * @param	string	the file name/path
   * @param	string	the data to be encoded
   * @return	void
   */
  function _add_data($filepath, $data, $file_mtime, $file_mdate) {
    $filepath = str_replace("\\", "/", $filepath);

    $uncompressed_size = strlen($data);
    $crc32 = crc32($data);

    $gzdata = gzcompress($data);
    $gzdata = substr($gzdata, 2, -4);
    $compressed_size = strlen($gzdata);

    $this->zipdata .=
            "\x50\x4b\x03\x04\x14\x00\x00\x00\x08\x00"
            . pack('v', $file_mtime)
            . pack('v', $file_mdate)
            . pack('V', $crc32)
            . pack('V', $compressed_size)
            . pack('V', $uncompressed_size)
            . pack('v', strlen($filepath)) // length of filename
            . pack('v', 0) // extra field length
            . $filepath
            . $gzdata; // "file data" segment

    $this->directory .=
            "\x50\x4b\x01\x02\x00\x00\x14\x00\x00\x00\x08\x00"
            . pack('v', $file_mtime)
            . pack('v', $file_mdate)
            . pack('V', $crc32)
            . pack('V', $compressed_size)
            . pack('V', $uncompressed_size)
            . pack('v', strlen($filepath)) // length of filename
            . pack('v', 0) // extra field length
            . pack('v', 0) // file comment length
            . pack('v', 0) // disk number start
            . pack('v', 0) // internal file attributes
            . pack('V', 32) // external file attributes - 'archive' bit set
            . pack('V', $this->offset) // relative offset of local header
            . $filepath;

    $this->offset = strlen($this->zipdata);
    $this->entries++;
    $this->file_num++;
  }

  // --------------------------------------------------------------------

  /**
   * Read the contents of a file and add it to the zip
   *
   * @access	public
   * @return	bool
   */
  function read_file($path, $preserve_filepath = FALSE) {
    if (!file_exists($path)) {
      return FALSE;
    }

    if (FALSE !== ($data = file_get_contents($path))) {
      $name = str_replace("\\", "/", $path);

      if ($preserve_filepath === FALSE) {
        $name = preg_replace("|.*/(.+)|", "\\1", $name);
      }

      $this->add_data($name, $data);
      return TRUE;
    }
    return FALSE;
  }

  // ------------------------------------------------------------------------

  /**
   * Read a directory and add it to the zip.
   *
   * This function recursively reads a folder and everything it contains (including
   * sub-folders) and creates a zip based on it.  Whatever directory structure
   * is in the original file path will be recreated in the zip file.
   *
   * @access	public
   * @param	string	path to source
   * @return	bool
   */
  function read_dir($path, $preserve_filepath = TRUE, $root_path = NULL) {
    if (!$fp = @opendir($path)) {
      return FALSE;
    }

    // Set the original directory root for child dir's to use as relative
    if ($root_path === NULL) {
      $root_path = dirname($path) . '/';
    }

    while (FALSE !== ($file = readdir($fp))) {
      if (substr($file, 0, 1) == '.') {
        continue;
      }

      if (@is_dir($path . $file)) {
        $this->read_dir($path . $file . "/", $preserve_filepath, $root_path);
      } else {
        if (FALSE !== ($data = file_get_contents($path . $file))) {
          $name = str_replace("\\", "/", $path);

          if ($preserve_filepath === FALSE) {
            $name = str_replace($root_path, '', $name);
          }

          $this->add_data($name . $file, $data);
        }
      }
    }

    return TRUE;
  }

  // --------------------------------------------------------------------

  /**
   * Get the Zip file
   *
   * @access	public
   * @return	binary string
   */
  function get_zip() {
    // Is there any data to return?
    if ($this->entries == 0) {
      return FALSE;
    }

    $zip_data = $this->zipdata;
    $zip_data .= $this->directory . "\x50\x4b\x05\x06\x00\x00\x00\x00";
    $zip_data .= pack('v', $this->entries); // total # of entries "on this disk"
    $zip_data .= pack('v', $this->entries); // total # of entries overall
    $zip_data .= pack('V', strlen($this->directory)); // size of central dir
    $zip_data .= pack('V', strlen($this->zipdata)); // offset to start of central dir
    $zip_data .= "\x00\x00"; // .zip file comment length

    return $zip_data;
  }

  // --------------------------------------------------------------------

  /**
   * Write File to the specified directory
   *
   * Lets you write a file
   *
   * @access	public
   * @param	string	the file name
   * @return	bool
   */
  function archive($filepath) {
    if (!($fp = @fopen($filepath, FOPEN_WRITE_CREATE_DESTRUCTIVE))) {
      return FALSE;
    }

    flock($fp, LOCK_EX);
    fwrite($fp, $this->get_zip());
    flock($fp, LOCK_UN);
    fclose($fp);

    return TRUE;
  }

  // --------------------------------------------------------------------

  /**
   * Download
   *
   * @access	public
   * @param	string	the file name
   * @param	string	the data to be encoded
   * @return	bool
   */
  function download($filename = 'backup.zip') {
    if (!preg_match("|.+?\.zip$|", $filename)) {
      $filename .= '.zip';
    }

    $CI = & get_instance();
    $CI->load->helper('download');

    $get_zip = $this->get_zip();

    $zip_content = & $get_zip;

    force_download($filename, $zip_content);
  }

  // --------------------------------------------------------------------

  /**
   * Initialize Data
   *
   * Lets you clear current zip data.  Useful if you need to create
   * multiple zips with different data.
   *
   * @access	public
   * @return	void
   */
  function clear_data() {
    $this->zipdata = '';
    $this->directory = '';
    $this->entries = 0;
    $this->file_num = 0;
    $this->offset = 0;
  }

}

/* End of file Zip.php */
/* Location: ./system/libraries/Zip.php */